nvidia驱动、cuda和GPU版本pytorch安装 |
您所在的位置:网站首页 › pytorch 显卡型号 › nvidia驱动、cuda和GPU版本pytorch安装 |
1、nvidia官网:https://www.nvidia.cn/ 2、找到 驱动程序 -> 所有 NVIDIA驱动程序 找到对应显卡的最新驱动 3、根据自己电脑的显卡型号 查找驱动,如 GTX1050
4、找到后,下载安装即可 问题:NVIDIA-SMI has failed because you are not: a) running as an administrator or b) there is not at least on ......... 原因:因Win10 系统更新导致的新的内核 与 nvidia驱动不匹配,导致的 GPU不工作 重装驱动后问题解决: 在 命令提示符窗口 cd C:\Program Files\NVIDIA Corporation\NVSMI ,然后输入 nvidai-smi 会输出显卡与驱动信息,若驱动程序不匹配,则会报错信息。 cuda 和 cudnn安装 5、cuda 下载地址:cuda 根据自己笔记本 选择相应配置: 6、cuDNN下载地址: https://developer.nvidia.com/rdp/cudnn-archive 根据步骤5下载的 cuda 版本下载对应的 cudnn即可 7、下载安装对应GPU版本的 pytorch 即可 pytorch 下载地址:https://pytorch.org/get-started/locally/
Reference: GPU版pytorch安装
|
今日新闻 |
推荐新闻 |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |